# -*- coding: utf-8 -*-
from django.contrib import admin

from helpdesk.models import Ocorrencia, Categoria, Tecnico, OcorrenciaStatus
from helpdesk.forms import OcorrenciaForm
from djtools.templatetags.djtools_templatetags import in_group

admin.site.register(Categoria)
admin.site.register(Tecnico)
admin.site.register(OcorrenciaStatus)
#admin.site.register(Ocorrencia)


##############
# OCORRENCIA #
##############

class OcorrenciaAdmin(admin.ModelAdmin):
    form = OcorrenciaForm
    search_fields = ['numero_patrimonio', 'status']
    list_display = ['editar', 'status', 'setor', 'numero_patrimonio', 'get_data_chegada', 'usuario', 'problema']
    list_filter = ['status']
    fieldsets = [
        ('', {'fields': ['status','categoria','data_chegada','data_saida',]}),
        (u'Localização', {'fields': ['setor','numero_patrimonio','usuario']}),
        (u'Ocorrência', {'fields': ['ocorretor','tecnico','problema','observacao']}),
        ('', {'fields': ['enviar']}),
    ]
    
    def queryset(self, request): #TODO: Usado para filtrar
#        if in_group(request.user, ['helpdesk_tecnico']):
#        else:
        return Ocorrencia.objects.all()
    
    def get_data_chegada(self, obj):
        return obj.data_chegada.strftime("%d/%m/%Y")
    get_data_chegada.short_description = u'Data de Chegada'
    get_data_chegada.admin_order_field = 'data_chegada'
    get_data_chegada.allow_tags = True

    def editar(self, obj):
        return '<img src="/media/img/16x16/editar.png" title="Editar" />'
    editar.allow_tags = True
    editar.short_description = ''
    editar.attrs = {'width': '10px'}
admin.site.register(Ocorrencia, OcorrenciaAdmin)